# log_util.py
import logging
import os
from logging.handlers import RotatingFileHandler

# 在当前脚本所在目录下创建日志文件
current_dir = os.path.dirname(os.path.abspath(__file__))
log_dir_path = current_dir

# 创建日志记录器，指明日志保存路径,每个日志的大小，保存日志的上限
file_log_handler = RotatingFileHandler(os.path.join(log_dir_path, 'log.log'), maxBytes=1024 * 1024, backupCount=10,
                                       encoding="utf-8")

# 设置日志的格式
date_string = '%Y-%m-%d %H:%M:%S'
formatter = logging.Formatter(
    '[%(asctime)s] [%(levelname)s] [%(filename)s]/[line: %(lineno)d]/[%(funcName)s] %(message)s ', date_string)

# 日志输出到控制台的句柄
stream_handler = logging.StreamHandler()

# 将日志记录器指定日志的格式
file_log_handler.setFormatter(formatter)
stream_handler.setFormatter(formatter)

# 创建logger实例
logger = logging.getLogger(__name__)

# 清空现有处理器，避免重复添加
for handler in logger.handlers[:]:
    logger.removeHandler(handler)

# 为全局的日志工具对象添加日志记录器
logger.addHandler(stream_handler)
logger.addHandler(file_log_handler)

# 设置日志输出级别
logger.setLevel(level=logging.DEBUG)
